from fastapi import APIRouter, Depends, Security
from fastapi.responses import JSONResponse

from myProfile.models.data.myProfile_models import Login, PMH
from myProfile.repository.personal_medical_history import PMH_Repository
from myProfile.models.request.myProfile_request import PMHReq
from myProfile.db_config.odmantic_config import create_db_engine
from myProfile.security.secure import get_current_valid_user

router = APIRouter()


@router.post("/pmh/add")
async def add_pmh(req: PMHReq,
                  current_user: Login = Security(
                      get_current_valid_user, scopes=["user"]),
                  engine=Depends(create_db_engine)):
    pmh = PMH(**req.model_dump())
    repo: PMH_Repository = PMH_Repository(engine)
    result = await repo.add_to_profile(current_user.id, pmh)
    if result is True:
        return req
    else:
        return JSONResponse(
            content={"message": "add pmh problem encountered."},
            status_code=500)


@router.delete("/pmh/delete/{id}")
async def delete_pmh(id: int,
                     current_user: Login = Security(
                         get_current_valid_user, scopes=["user"]),
                     engine=Depends(create_db_engine)):
    repo: PMH_Repository = PMH_Repository(engine)
    result = await repo.remove_from_profile(current_user.id, id)
    if result is True:
        return JSONResponse(
            content={"message": "delete pmh successfully"},
            status_code=201)
    else:
        return JSONResponse(
            content={"message": "delete pmh unsuccessfully"},
            status_code=500)


@router.get("/pmh/get")
async def get_pmh(current_user: Login = Security(
                    get_current_valid_user, scopes=["user"]),
                  engine=Depends(create_db_engine)):
    repo: PMH_Repository = PMH_Repository(engine)
    return await repo.get_from_profile(current_user.id)
